package com.welink.cmp.convertor;

import org.springframework.beans.BeanUtils;

import java.util.List;
import java.util.stream.Collectors;

public class GenericConvertor {

    /**
     * 将 DO 转换为 CO
     *
     * @param source DO 对象
     * @param targetClass CO 类
     * @param <S> 源类型（DO 类型）
     * @param <T> 目标类型（CO 类型）
     * @return 转换后的 CO 对象
     */
    public static <S, T> T toCO(S source, Class<T> targetClass) {
        if (source == null) {
            return null;
        }
        try {
            T target = targetClass.getDeclaredConstructor().newInstance();
            BeanUtils.copyProperties(source, target);
            return target;
        } catch (Exception e) {
            throw new RuntimeException("Conversion failed", e);
        }
    }

    /**
     * 将 CO 转换为 DO
     *
     * @param source CO 对象
     * @param targetClass DO 类
     * @param <S> 源类型（CO 类型）
     * @param <T> 目标类型（DO 类型）
     * @return 转换后的 DO 对象
     */
    public static <S, T> T toDO(S source, Class<T> targetClass) {
        if (source == null) {
            return null;
        }
        try {
            T target = targetClass.getDeclaredConstructor().newInstance();
            BeanUtils.copyProperties(source, target);
            return target;
        } catch (Exception e) {
            throw new RuntimeException("Conversion failed", e);
        }
    }

    /**
     * 将 List<DO> 转换为 List<CO>
     *
     * @param sourceList DO 对象列表
     * @param targetClass CO 类
     * @param <S> 源类型（DO 类型）
     * @param <T> 目标类型（CO 类型）
     * @return 转换后的 CO 对象列表
     */
    public static <S, T> List<T> toCOList(List<S> sourceList, Class<T> targetClass) {
        if (sourceList == null) {
            return null;
        }
        return sourceList.stream()
                .map(source -> toCO(source, targetClass))
                .collect(Collectors.toList());
    }

    /**
     * 将 List<CO> 转换为 List<DO>
     *
     * @param sourceList CO 对象列表
     * @param targetClass DO 类
     * @param <S> 源类型（CO 类型）
     * @param <T> 目标类型（DO 类型）
     * @return 转换后的 DO 对象列表
     */
    public static <S, T> List<T> toDOList(List<S> sourceList, Class<T> targetClass) {
        if (sourceList == null) {
            return null;
        }
        return sourceList.stream()
                .map(source -> toDO(source, targetClass))
                .collect(Collectors.toList());
    }
}